package util;

import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.ext.hsqldb.HsqldbDataTypeFactory;

import java.sql.Connection;

public class CustomDatabaseConnection extends DatabaseConnection {
    public CustomDatabaseConnection(Connection connection, String schema) {
        super(connection, schema);
    }

    public CustomDatabaseConnection(Connection connection) {
        super(connection);
        this.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new HsqldbDataTypeFactory());
    }
}
